#!/usr/bin/env -S uv run
# /// script
# requires-python = ">=3.10"
# dependencies = [
#     "zarr@git+https://github.com/zarr-developers/zarr-python.git@main",
#     "numpy",
# ]
# ///

import os

import numpy as np
import zarr

# Create array
store = zarr.open_group("test.zarr", mode="w")
arr = store.create(
    "data",
    shape=(20,),
    chunks=(10,),
    dtype="f8",
    fill_value=0.0,
    config={"write_empty_chunks": True},
)

# Reopen (like xarray does with regions)
store = zarr.open_group("test.zarr", mode="r+")
arr = zarr.open(store=store.store, path="data", config={"write_empty_chunks": True})

# Write zeros - should create chunk but doesn't
arr[0:10] = np.zeros(10)

# Check what was written
chunk_dir = "test.zarr/data/c"
chunks = (
    [f for f in os.listdir(chunk_dir) if f.isdigit()]
    if os.path.exists(chunk_dir)
    else []
)
print(f"Chunks written: {sorted(chunks)}")
print("Expected: ['0'] (chunk with zeros should be written)")